#include <led-drivers/net/DriverNetUdpH801.h>

// Constants
namespace {

	const ushort H801_DEFAULT_PORT = 30977;
	const char H801_DEFAULT_HOST[] = "255.255.255.255";

} //End of constants

DriverNetUdpH801::DriverNetUdpH801(const QJsonObject& deviceConfig)
	: ProviderUdp(deviceConfig)
{
}

bool DriverNetUdpH801::init(QJsonObject deviceConfig)
{
	bool isInitOK = false;

	/* The H801 port is fixed */
	_port = H801_DEFAULT_PORT;
	_defaultHost = H801_DEFAULT_HOST;

	// Initialise sub-class
	if (ProviderUdp::init(deviceConfig))
	{
		_ids.clear();
		QJsonArray lArray = deviceConfig["lightIds"].toArray();
		for (int i = 0; i < lArray.size(); i++)
		{
			QString id = lArray[i].toString();
			_ids.push_back(id.toInt(nullptr, 16));
		}

		_message = QByteArray(_prefix_size + _colors + _id_size * _ids.size() + _suffix_size, 0x00);
		_message[0] = static_cast<char>(0xFB);
		_message[1] = static_cast<char>(0xEB);

		for (int i = 0; i < _ids.length(); i++) {
			_message[_prefix_size + _colors + i * _id_size + 0] = (_ids[i] >> 0x00) & 0xFF;
			_message[_prefix_size + _colors + i * _id_size + 1] = (_ids[i] >> 0x08) & 0xFF;
			_message[_prefix_size + _colors + i * _id_size + 2] = (_ids[i] >> 0x10) & 0xFF;
		}

		Debug(_log, "H801 using %s:%d", _address.toString().toStdString().c_str(), _port);

		isInitOK = true;
	}
	return isInitOK;
}

int DriverNetUdpH801::writeFiniteColors(const std::vector<ColorRgb>& ledValues)
{
	ColorRgb color = ledValues[0];
	_message[_prefix_size + 0] = color.red;
	_message[_prefix_size + 1] = color.green;
	_message[_prefix_size + 2] = color.blue;

	return writeBytes(_message.size(), reinterpret_cast<const uint8_t*>(_message.data()));
}

LedDevice* DriverNetUdpH801::construct(const QJsonObject& deviceConfig)
{
	return new DriverNetUdpH801(deviceConfig);
}

bool DriverNetUdpH801::isRegistered = hyperhdr::leds::REGISTER_LED_DEVICE("udph801", "leds_group_2_network", DriverNetUdpH801::construct);
